SegNet implementation in PyTorch

Move the frame to colab [Done]

Mount data [done]

Fix the training function [Done]

Report training accuracy [Done]

Learning Rate decay and cooldown [Done]

Save checkpints [Done]

Visualize an output [Done]

Plot training curve [Done]

Calculate class weights median frequency balancing [Done]

Online Data Augmentation (intensity, noise, stretch, center shift) [Done]

Quantitive evaluation (accuracy, etc) [Done]

Video visualization [Done]

Demo on CamVid Seq01, Kitti(preprocessing?) [Done]

Global vars

Reporting/Evaluation Functions

Save/Load checkpoints

Visualization

Dataset Preparation

Train

Train from scratch

Train from VGG weights

Load model

Report Evaluation results on test dataset

Visualization Output

Demostration

On Demo.py

Overlay Segmentation results